FROM qwenllm/qwenvl:2.5-cu121

ENV FORCE_CUDA="1"

WORKDIR /workspace

RUN apt-get update && apt-get install libsm6 libxext6 libgl1-mesa-glx -y
RUN apt-get install -y --no-install-recommends ffmpeg python-is-python3

COPY pyproject.toml .
COPY uv.lock .

RUN pip install uv
RUN uv pip install --system --no-build-isolation "flash-attn==2.7.4.post1" 
RUN uv pip install --system . 
